#pragma once

#include <iostream>
#include <signal.h>
#include <functional>
#include "Sock.hpp"

class HttpServer
{
public:
    using func_t = std::function<void(int)>;
private: // 成员变量
    int _listensock;
    uint16_t _port;
    Sock sock;
    func_t _funcs;

public: // 默认成员函数
    HttpServer(const uint16_t& port, const func_t& func)
        : _port(port), _funcs(func)
    {
        // 创建套接字 + 初始化
        _listensock = sock.Socket();
        sock.Bind(_listensock, _port);
        sock.Listen(_listensock);
    }

    ~HttpServer()
    {}

public: // 成员函数
    void Start()
    {   
        // 子进程终止或停止时发送给父进程的信号
        signal(SIGCHLD, SIG_IGN); // SIGCHLD: 子进程终止或停止时发送给父进程的信号
        while(1)
        {
            std::string client_ip;
            uint16_t client_port = 0;
            int sockfd = sock.Accept(_listensock, &client_ip, &client_port);
            if(sockfd < 0)
            {
                std::cerr << "Accept error!" << std::endl;
                continue;
            }
            
            if(fork() == 0)
            {
                // 子进程
                close(_listensock);
                _funcs(sockfd);
                close(sockfd);
            }
        }
    }

};